# Python 单元测试

# 导入 unittest 模块
from unittest import TestCase

# 导入被测试模块
from calculator import Calculator


# 定义测试用例继承自 TestCase
class TestCalculator(TestCase):

    # 测试加法
    def test_add(self):
        c = Calculator()
        # 断言计算结果与期望结果为 5
        self.assertEqual(c.add(2, 3), 5)

    # 测试减法
    def test_sub(self):
        c = Calculator()
        self.assertEqual(c.subtract(5, 3), 3)

    # 测试乘法
    def test_multiply(self):
        c = Calculator()
        self.assertEqual(c.multiply(3, 9), 27)

    # 测试除法
    def test_divide(self):
        c = Calculator()
        self.assertEqual(c.divide(45, 5), 12)